package com.ruoyi.app.utils;

import ws.schild.jave.*;

import java.io.File;
import java.math.BigDecimal;

/**
 * @author 侯存路
 * @description
 * @since 2020/8/20 10:48
 */
public class VideoUtil {


    /**
     * 视频文件压缩
     *
     * @param sourceFile
     * @return
     */
    public static File compressedVideo(File sourceFile) {
        String sourcePath = sourceFile.getAbsolutePath();
        String rootPath = sourcePath.substring(0, sourcePath.lastIndexOf(File.separator));
        String newFilePath = rootPath + File.separator + "new-" + sourceFile.getName();

        System.out.println("原文件路径" + sourcePath);
        System.out.println("压缩文件路径" + newFilePath);
        File target = new File(newFilePath);
        try {
            MultimediaObject object = new MultimediaObject(sourceFile);
            AudioInfo audioInfo = object.getInfo().getAudio();
            // 根据视频大小来判断是否需要进行压缩,
            int maxSize = 100;
            double mb = Math.ceil(sourceFile.length() / 1048576);
            int second = (int) object.getInfo().getDuration() / 1000;
            BigDecimal bd = new BigDecimal(String.format("%.4f", mb / second));
            System.out.println("开始压缩视频了--> 视频每秒平均 " + bd + " MB ");
            // 视频 > 100MB, 或者每秒 > 0.5 MB 才做压缩， 不需要的话可以把判断去掉
            boolean temp = mb > maxSize || bd.compareTo(new BigDecimal(0.5)) > 0;
            if (temp) {
                long time = System.currentTimeMillis();
                Integer maxBitRate = 228000;
                int maxSamplingRate = 84100;
                int bitRate = 1600000;
                int maxFrameRate = 40;
                int maxWidth = 2280;
                AudioAttributes audio = new AudioAttributes();
                // 设置通用编码格式
                audio.setCodec("aac");
                // 设置最大值：比特率越高，清晰度/音质越好
                // 设置音频比特率,单位:b (比特率越高，清晰度/音质越好，当然文件也就越大 128000 = 182kb)
                if (audioInfo.getBitRate() > maxBitRate) {
                    audio.setBitRate(maxBitRate);
                }
                // 设置重新编码的音频流中使用的声道数（1 =单声道，2 = 双声道（立体声））。如果未设置任何声道值，则编码器将选择默认值 0。
                audio.setChannels(audioInfo.getChannels());
                // 采样率越高声音的还原度越好，文件越大
                // 设置音频采样率，单位：赫兹 hz
                // 设置编码时候的音量值，未设置为0,如果256，则音量值不会改变
                // audio.setVolume(256);
                if (audioInfo.getSamplingRate() > maxSamplingRate) {
                    audio.setSamplingRate(maxSamplingRate);
                }
                VideoInfo videoInfo = object.getInfo().getVideo();
                VideoAttributes video = new VideoAttributes();
                video.setCodec("libx264");
                //设置音频比特率,单位:b (比特率越高，清晰度/音质越好，当然文件也就越大 800000 = 800kb)
                if (videoInfo.getBitRate() > bitRate) {
                    video.setBitRate(bitRate);
                }
                // 视频帧率：15 f / s  帧率越低，效果越差
                // 设置视频帧率（帧率越低，视频会出现断层，越高让人感觉越连续），视频帧率（Frame rate）是用于测量显示帧数的量度。所谓的测量单位为每秒显示帧数(Frames per Second，简：FPS）或“赫兹”（Hz）。
                if (videoInfo.getFrameRate() > maxFrameRate) {
                    video.setFrameRate(maxFrameRate);
                }
                // 限制视频宽高
                int width = videoInfo.getSize().getWidth();
                int height = videoInfo.getSize().getHeight();
                if (width > maxWidth) {
                    float rat = (float) width / maxWidth;
                    video.setSize(new VideoSize(maxWidth, (int) (height / rat)));
                }
                EncodingAttributes attr = new EncodingAttributes();
                attr.setFormat("mp4");
                attr.setAudioAttributes(audio);
                attr.setVideoAttributes(video);
//                // 设置线程数
                System.out.println("线程数######" + Runtime.getRuntime().availableProcessors());
                attr.setEncodingThreads(Runtime.getRuntime().availableProcessors());
                attr.setDecodingThreads(Runtime.getRuntime().availableProcessors());
                System.out.println("压缩总耗时1：" + (System.currentTimeMillis() - time) / 1000);
                Encoder encoder = new Encoder();
                encoder.encode(new MultimediaObject(sourceFile), target, attr);
                System.out.println("压缩总耗时：" + (System.currentTimeMillis() - time) / 1000);
                return target;
            } else {
                return sourceFile;
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }


    public static void main(String[] args) {

        File file = new File("G:\\test.mp4");

        File newFile = compressedVideo(file);

    }


}
